/*
 * Copyright (C) 2005-2011 MaNGOS <http://www.getmangos.com/>
 *
 * Copyright (C) 2008-2011 Trinity <http://www.trinitycore.org/>
 *
 * Copyright (C) 2010-2011 ProjectSkyfire <http://www.projectskyfire.org/>
 * 
 * Copyright (C) 2011 ArkCORE <http://www.arkania.net/>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#ifndef _CALLBACK_H
#define _CALLBACK_H

#include <ace/Future.h>
#include <ace/Future_Set.h>
#include "QueryResult.h"

typedef ACE_Future<QueryResult> QueryResultFuture;
typedef ACE_Future<PreparedQueryResult> PreparedQueryResultFuture;

/*! A simple template using ACE_Future to manage callbacks from the thread and object that
 issued the request. <ParamType> is variable type of parameter that is used as parameter
 for the callback function.
 */
template<typename Result, typename ParamType>
class QueryCallback {
public:
	QueryCallback() {
	}

	void SetFutureResult(ACE_Future<Result> value) {
		result = value;
	}

	ACE_Future<Result> GetFutureResult() {
		return result;
	}

	int IsReady() {
		return result.ready();
	}

	void GetResult(Result& res) {
		result.get(res);
	}

	void FreeResult() {
		result.cancel();
	}

	void SetParam(ParamType value) {
		param = value;
	}

	ParamType GetParam() {
		return param;
	}

private:
	ACE_Future<Result> result;
	ParamType param;
};

template<typename Result, typename ParamType1, typename ParamType2>
class QueryCallback_2 {
public:
	QueryCallback_2() {
	}

	void SetFutureResult(ACE_Future<Result> value) {
		result = value;
	}

	ACE_Future<Result> GetFutureResult() {
		return result;
	}

	int IsReady() {
		return result.ready();
	}

	void GetResult(Result& res) {
		result.get(res);
	}

	void FreeResult() {
		result.cancel();
	}

	void SetFirstParam(ParamType1 value) {
		param_1 = value;
	}

	void SetSecondParam(ParamType2 value) {
		param_2 = value;
	}

	ParamType1 GetFirstParam() {
		return param_1;
	}

	ParamType2 GetSecondParam() {
		return param_2;
	}

private:
	ACE_Future<Result> result;
	ParamType1 param_1;
	ParamType2 param_2;
};

#endif
